import torch

#  创建一个 3D 张量 (例如：一个批次包含2个 3x4 的矩阵)
t = torch.randn(2,3,4)

# 形状
print(f"t.shape:{t.shape}")
print(f"t.size():{t.size()}")

# 获取特定维度的大小
print(f"Size of dimension 0: {t.shape[0]}")
print(f"Size of dimension 1: {t.shape[1]}")
print(f"Size of dimension 1: {t.shape[2]}")

# 维度
print(f"t.ndim: {t.ndim}")
print(f"t.dim(): {t.dim()}")